package com.jichangxiu.generator.config;

import com.jichangxiu.common.constant.Constants;
import org.apache.velocity.app.Velocity;

import java.util.Properties;

public class VelocityInitializer {

    /**
     * 初始化vm方法
     */
    public static void initVelocity() {
        Properties p = new Properties();
        try {
            // 加载 classpath 目录下的 vm 文件
            // p.setProperty("resource.loader.file.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");
            // 加载 String 类型的 vm 文件
            p.setProperty("resource.loader.file.class", "org.apache.velocity.runtime.resource.loader.StringResourceLoader");
            // 定义字符集
            p.setProperty(Velocity.INPUT_ENCODING, Constants.UTF_8);
            // 初始化 Velocity 引擎，指定配置 Properties
            Velocity.init(p);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

}
